import com.google.common.collect.HashMultimap;

import java.util.Set;

public class MainMenu {
    HashMultimap<String, Resource> options = HashMultimap.create();

    public HashMultimap<String, Resource> getOptions() {
        return options;
    }

    public void addNewBookToOption(Resource book) {
        String category = book.getOption();
        options.put(category, book);
    }

    public boolean hasThisCategory(String category) {
        return options.keys().contains(category);
    }

    public Set<String> getMainMenuName() {
        return options.keySet();
    }
}
